import pandas as pd
import SimpleITK as sitk


train_csv_dir = '../data/csv_label/train150_anno.csv'
val_csv_dir= '../data/csv_label/train51_anno.csv'

train_csv = pd.read_csv(train_csv_dir)
val_csv = pd.read_csv(val_csv_dir)
all_csv = pd.concat([train_csv,val_csv],axis=0)
all_csv = all_csv.sample(frac = 1)

all_csv.iloc[:100].to_csv('../data/csv_label/train_1.csv')
all_csv.iloc[100:].to_csv('../data/csv_label/val_1.csv')

all_csv.iloc[:50].to_csv('../data/csv_label/val_2.csv')
all_csv.iloc[50:].to_csv('../data/csv_label/train_2.csv')

